package com.backend.vse.ws;

import com.alibaba.fastjson.JSON;
import com.backend.vse.common.Result;
import com.backend.vse.entity.User;
import com.backend.vse.interceptor.JwtInterceptor;
import com.backend.vse.service.UserService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * WebSocket 控制器，用于处理实验相关操作。
 * 该控制器提供匹配用户和开始实验的端点。
 */
@Slf4j
@RestController
@RequestMapping("/exp8")
public class WsController {
    private static WsUser user = new WsUser(); // 静态变量，用于存储当前参与实验的用户。

    @Autowired
    private UserService userService; // 自动注入 UserService，用于与用户数据交互。

    /**
     * 匹配当前用户与另一个在线用户进行实验。
     *
     * @param index 当前用户的索引
     * @param role  当前用户的角色（例如，“代理人”或“委托人”）
     * @return 包含匹配用户数据或错误信息的结果
     */
    @GetMapping("/matchUser")
    public Result matchUser(String index, String role) {
        List<String> userIds = JwtInterceptor.getUserIds(); // 从拦截器获取在线用户ID列表。
        if (userIds != null && !userIds.isEmpty()) {
            // 过滤掉当前用户，找到另一个在线用户
            List<String> list = userIds.stream().filter(i -> !i.equals(index)).toList();
            if (!list.isEmpty()) {
                User suser = userService.findUserByIndex(Long.valueOf(index)); // 查找当前用户信息
                User user = userService.findUserByIndex(Long.valueOf(list.get(0))); // 查找匹配用户信息

                // 构建匹配用户数据
                WsUser data = new WsUser()
                        .setSource(role)
                        .setSourceName(suser.getName())
                        .setTarget(role.equals("代理人") ? "委托人" : "代理人")
                        .setSourceId(index).setTargetId(list.get(0))
                        .setTargetName(user.getName());

                // 发送匹配事件消息
                Map<String, Object> map = new HashMap<>();
                map.put("event", "MATCH");
                map.put("data", data);
                WsServerHandler.sendMessage(JSON.toJSONString(map));

                return Result.success(data); // 返回成功结果
            }
        }
        return Result.fail(500, "没有匹配到在线对手"); // 返回错误结果
    }

    /**
     * 开始实验的端点。
     *
     * @param data 包含实验数据的 WsUser 对象
     * @return 包含实验开始信息的结果
     */
    @GetMapping("/begin")
    public Result begin(@RequestBody WsUser data) {
        log.info("开始实验, {}", data); // 记录日志
        if (user.getSourceId() != null) {
            return Result.fail(500, "已经有用户在实验中"); // 如果已有用户在实验中，返回错误结果
        }
        user = data; // 设置当前实验用户
        return Result.success(data); // 返回成功结果
    }
}
